import os
from llama_cloud_services import LlamaParse
from llama_index.core import SimpleDirectoryReader


# 获取环境变量中的API密钥
api_key ="llx-TAOOZoEycuUtXqnszLYfWYvmsEMq0GWjL0pXiFZrRqrunC1V"

# 如果没有设置API密钥，给出明确的提示信息
if not api_key:
    raise ValueError("请先在 https://cloud.llamaindex.ai 注册并获取 API 密钥，然后设置环境变量 LLAMA_CLOUD_API_KEY")

# set up parser
parser = LlamaParse(
    result_type="markdown",  # "markdown" and "text" are available
    api_key=api_key
)
file_extractor = {".pdf": parser}

documents = SimpleDirectoryReader(input_dir="./data", required_exts=[".pdf"], file_extractor=file_extractor).load_data()
print(documents[0].text)